package com.whisper.common.utils;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.whisper.common.annotation.PageQuery;
import com.whisper.common.constant.HttpStatus;
import com.whisper.common.core.page.PageDomain;
import com.whisper.common.core.page.TableDataInfo;
import com.whisper.common.core.page.TableSupport;
import com.whisper.common.utils.sql.SqlUtil;

import java.util.Collection;
import java.util.List;

/**
 * 分页工具类
 * 
 * @author whisper
 */
public class PageUtils extends PageHelper
{

    public TableDataInfo startPage(Object returnValue, PageQuery page) {
        if (returnValue instanceof Collection) {
            if (page.returnAll()) {
                return returnAll((List<Object>) returnValue);
            } else {
                return returnPaging((List<Object>) returnValue);
            }
        }
        throw new UnsupportedOperationException("返回值异常！无法进行分页操作");
    }

    /**
     * 设置请求分页数据
     */
    public static void startPage()
    {
        PageDomain pageDomain = TableSupport.buildPageRequest();
        Integer pageNum = pageDomain.getPageNum();
        Integer pageSize = pageDomain.getPageSize();
        String orderBy = SqlUtil.escapeOrderBySql(pageDomain.getOrderBy());
        Boolean reasonable = pageDomain.getReasonable();
        PageHelper.startPage(pageNum, pageSize, orderBy).setReasonable(reasonable);
    }

    /**
     * 全查模式
     * @param list
     * @return
     */
    private TableDataInfo returnAll(List<Object> list) {
        TableDataInfo rspData = new TableDataInfo();
        rspData.setCode(HttpStatus.SUCCESS);
        rspData.setRows(list);
        rspData.setMsg("查询成功");
        rspData.setTotal(list.size());
        return rspData;
    }

    /**
     * 分页模式
     * @param list
     * @return
     */
    private TableDataInfo returnPaging(List<Object> list) {
        TableDataInfo rspData = new TableDataInfo();
        rspData.setCode(HttpStatus.SUCCESS);
        rspData.setRows(list);
        rspData.setMsg("查询成功");
        rspData.setTotal(new PageInfo(list).getTotal());
        return rspData;
    }

    /**
     * 清理分页的线程变量
     */
    public static void clearPage()
    {
        PageHelper.clearPage();
    }
}
